import os
import subprocess

import click
import requests


class PkgGitRepo(object):
    def __init__(self, repo_name, gitee_pat, gitee_org, gitee_user=None,
                 gitee_email=None):
        self.gitee_org = gitee_org
        self.repo_name = repo_name
        self.gitee_pat = gitee_pat
        self.gitee_user = gitee_user
        self.gitee_email = gitee_email
        self.not_found = False
        self.branch_not_found = False
        self.repo_dir = ''
        self.commit_pushed = False

    def fork_repo(self):
        try:
            url = "https://gitee.com/api/v5/repos/%s/%s/forks" % (
                self.gitee_org, self.repo_name)
            resp = requests.request("POST", url,
                                    data={"access_token": self.gitee_pat})
            if resp.status_code == 404:
                click.echo("Repo not found for: %s/%s" % (self.gitee_org,
                                                          self.repo_name),
                           err=True)
                self.not_found = True
            elif resp.status_code != 201:
                click.echo("Fork repo failed, %s" % resp.text, err=True)
        except requests.RequestException as e:
            click.echo("HTTP request to gitee failed: %s" % e, err=True)

    def clone_repo(self, src_dir):
        clone_url = "https://gitee.com/%s/%s" % (
            self.gitee_user, self.repo_name)
        click.echo("Cloning source repo from: %s" % clone_url)
        repo_dir = os.path.join(src_dir, self.repo_name)
        if os.path.exists(repo_dir):
            subprocess.call(["rm", "-fr", repo_dir])
        subprocess.call(["git", "clone", clone_url, repo_dir])
        self.repo_dir = os.path.join(src_dir, self.repo_name)

    def add_branch(self, src_branch, dest_branch):
        url = "https://gitee.com/api/v5/repos/{gitee_org}/{repo_name}/" \
              "branches/{dest_branch}".format(gitee_org=self.gitee_org,
                                              repo_name=self.repo_name,
                                              dest_branch=dest_branch)
        resp = requests.request("GET", url)
        if resp.status_code == 404:
            click.echo("Branch: %s not found for project: %s/%s" %
                       (dest_branch, self.gitee_org, self.repo_name),
                       err=True)
            self.branch_not_found = True
            return
        click.echo("Adding branch for %s/%s" % (self.gitee_org, self.repo_name))
        cmd = 'cd %(repo_dir)s; ' \
              'git config --global user.email "%(gitee_email)s";' \
              'git config --global user.name "%(gitee_user)s";' \
              'git remote add upstream "https://gitee.com/%(gitee_org)s/' \
              '%(repo_name)s";' \
              'git remote update;' \
              'git checkout -b %(src_branch)s upstream/%(dest_branch)s; ' % {
                  "repo_dir": self.repo_dir,
                  "gitee_user": self.gitee_user,
                  "gitee_email": self.gitee_email,
                  "gitee_org": self.gitee_org,
                  "repo_name": self.repo_name,
                  "src_branch": src_branch,
                  "dest_branch": dest_branch}
        click.echo("CMD: %s" % cmd)
        subprocess.call(cmd, shell=True)

    def commit(self, commit_message, do_push=True):
        click.echo("Commit changes for %s/%s" % (
            self.gitee_org, self.repo_name))
        commit_cmd = 'cd %(repo_dir)s/; ' \
                     'git add .; ' \
                     'git commit -am "%(commit_message)s";' \
                     'git remote set-url origin https://%(gitee_user)s:' \
                     '%(gitee_pat)s@gitee.com/%(gitee_user)s/%(repo_name)s;' \
                     % {"repo_dir": self.repo_dir,
                        "repo_name": self.repo_name,
                        "gitee_user": self.gitee_user,
                        "gitee_pat": self.gitee_pat,
                        "commit_message": commit_message}
        if do_push:
            commit_cmd += 'git push origin -f'
            self.commit_pushed = True
        click.echo("CMD: %s" % commit_cmd)
        subprocess.call(commit_cmd, shell=True)

    def create_pr(self, src_branch, remote_branch, tittle):
        if not self.commit_pushed:
            click.secho("WARNING: Commit was not pushed of %s, exit!" %
                        self.repo_name, fg='red')
            return
        click.echo("Creating pull request for project: %s" % self.repo_name)
        try:
            url = "https://gitee.com/api/v5/repos/%s/%s/pulls" % (
                self.gitee_org, self.repo_name)
            resp = requests.request(
                "POST", url, data={"access_token": self.gitee_pat,
                                   "title": tittle,
                                   "head": self.gitee_user + ":" + src_branch,
                                   "base": remote_branch})
            if resp.status_code != 201:
                click.echo("Create pull request failed, %s" % resp.text)
        except requests.RequestException as e:
            click.echo("HTTP request to gitee failed: %s" % e, err=True)

    def delete_fork(self):
        url = 'https://gitee.com/api/v5/repos/%s/%s?access_token=%s' % (
            self.gitee_user, self.repo_name, self.gitee_pat)
        resp = requests.request("DELETE", url)
        if resp.status_code == 404:
            click.echo("Repo %s/%s not found" % (
                self.gitee_user, self.repo_name))

    def pr_add_comment(self, comment, pr_num):
        click.echo("Adding comment: %s for project: %s in PR: %s" % (
            comment, self.repo_name, pr_num))
        url = 'https://gitee.com/api/v5/repos/%s/%s/pulls/%s/comments' % (
            self.gitee_org, self.repo_name, pr_num)
        body = {"access_token": "%s" % self.gitee_pat,
                "body": "%s" % comment}
        resp = requests.request("POST", url, data=body)
        if resp.status_code != 201:
            click.echo("Comment PR %s failed, reason: %s" %
                       (pr_num, resp.reason), err=True)
